package com.xx.gen1;

import com.baomidou.mybatisplus.extension.service.IService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.topsoft.topdp.mybatis.entity.StringIdExtendEntity;
import com.topsoft.topdp.mybatis.mapper.TopDpBaseMapper;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * <p>
 * 快速生成
 * </p>
 *
 * @author xuyanqing
 * @since 2022-11-15 14:28:30
 */
public class FastAutoGeneratorTest /*extends BaseGeneratorTest*/ {

    /**
     * 生成的文件存放路径
     * 原文链接：https://blog.csdn.net/TimerBin/article/details/127799360
     * <p>
     * System.getProperty("user.dir"); 获取当前模块所在的路径
     */
    private static final String FILE_STORAGE_FILE_ROOT_PATH = System.getProperty("user.dir") + "\\mybatis-plus-generator\\src\\main\\java";
    /**
     * 生成的mapper.xml 文件存放地址
     */
    private static final String FILE_STORAGE_MAPPER_XML_FILE_JAR_PACKAGE = System.getProperty("user.dir") + "\\mybatis-plus-generator\\src\\main\\resources\\mapper\\";
    /**
     * 数据源配置
     */
    private static final DataSourceConfig.Builder DATA_SOURCE_CONFIG = new DataSourceConfig
            .Builder("jdbc:postgresql://192.168.3.218:5432/top_gov", "top_gov", "top_gov").schema("public");
            //.Builder("jdbc:postgresql://192.168.3.248:5432/topdpmda", "postgres", "postgres").schema("public");
    //.Builder("jdbc:postgresql://192.168.3.246:5432/top_portal", "top_portal", "123456").schema("public");
            //.Builder("jdbc:postgresql://192.168.3.248:5432/five_portal", "postgres", "postgres").schema("public");

    /**
     * 执行 run
     */
    public static void main(String[] args) {
        FastAutoGenerator.create(DATA_SOURCE_CONFIG)
                .globalConfig(builder -> {
                    // 设置作者
                    builder.author("")
                            // 开启 swagger 模式
                            .enableSwagger()
                            // 覆盖已生成文件
                            // 禁止打开输出目录
                            .disableOpenDir()
                            // 指定输出目录
                            .outputDir(FILE_STORAGE_FILE_ROOT_PATH)
                            .fileOverride()
                            // 时间策略
                            .dateType(DateType.TIME_PACK)
                            // 时间格式
                            .commentDate("yyyy-MM-dd");
                })

                // https://blog.csdn.net/zhangHP_123/article/details/126061630
                .packageConfig(builder -> {
                    // 设置父包名
                    builder.parent("com.topsoft.stat")
                            // 设置父包模块名
                            .moduleName("warehousing")
                            // 实体类包名
                            .entity("entity")
                            // service包名
                            .service("service")
                            // serviceImpl包名
                            .serviceImpl("service.impl")
                            .mapper("dao")
                            // 指定xml生成的路径
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, FILE_STORAGE_MAPPER_XML_FILE_JAR_PACKAGE));
                })
                .strategyConfig(builder -> {
                    // 设置需要生成的表名
                    builder.addInclude("stat_project_audit_record")
                            .addTablePrefix("stat_", "c_", "mda_","aaf_")
                            // 自定义配置继承父包 HmBaseMapper  此类为我们公司扩展类
                            .mapperBuilder()
                            .superClass(TopDpBaseMapper.class)
                            .enableBaseColumnList()
                            .enableBaseResultMap()
                            // 如果不设置，如表[sys_user]，默认的文件名是[SysUserMapper]。写成下面这种形式后，将变成[SysUserDao]
                            .formatMapperFileName("%sDao")
                            .build()

                            // 实体类配置
                            .entityBuilder()
                            // 会在实体类前添加 [@Getter] 和 [@Setter] 注解。（这是Lombok的注解，需要添加Lombok依赖）
                            .enableLombok()
                            .enableTableFieldAnnotation()
//                            .enableChainModel()
                            .superClass(StringIdExtendEntity.class)
                            // 数据库表字段映射到实体的命名策略，下划线转驼峰命名
                            .columnNaming(NamingStrategy.underline_to_camel)
                            // 这些字段不会出现在新增的实体类中。
                            .addSuperEntityColumns("id", "deleted", "row_state", "update_time", "description", "create_user_id", "update_user_id")
                            // 会在实体类前添加 [@Accessors(chain = true)] 注解。用法如 [User user=new User().setAge(31).setName("snzl");]（这是Lombok的注解，需要添加Lombok依赖）
                            .enableChainModel()
                            .enableTableFieldAnnotation()
//                            .disableSerialVersionUID()
                            .build()

                            // service配置
                            .serviceBuilder()
//                            .superServiceClass(TopDpBaseService.class)
                            .superServiceClass(IService.class)
//                            .superServiceImplClass(TopDpBaseServiceImpl.class)
                            .superServiceImplClass(ServiceImpl.class)

                            // controller配置
                            .controllerBuilder()
                            .enableRestStyle()
                            .enableHyphenStyle()
//                            .convertFileName()
                            .build();

                })
                .injectionConfig(consumer -> {
                    Map<String, String> customFileMap = new HashMap<>(4);
                    // DTO 下面的key会作为类名后缀，进而生成新类
                    customFileMap.put("Dto.java", "templates/other/dto.java.ftl");
                    customFileMap.put("Vo.java", "templates/other/vo.java.ftl");
//                    customFileMap.put("dto"+ File.separator+"%s.java", "templates/other/dto.java.ftl");
//                    customFileMap.put("vo"+File.separator+"%s.java", "templates/other/vo.java.ftl");
//                    customFileMap.put("Convertor.java", "templates/other/convertor.java.ftl");
                    consumer.customFile(customFileMap);
                })
                // 使用Freemarker引擎模板，默认的是Velocity引擎模板
//                .templateEngine(new FreemarkerTemplateEngine())
                // 解决DTO,VO类生成的路径和文件名问题
                .templateEngine(new EnhanceFreemarkerTemplateEngine())
                .execute();

    }


    // 初始化数据库脚本
//        initDataSource(DATA_SOURCE_CONFIG.build());
        /*DATA_SOURCE_CONFIG.build();
        FastAutoGenerator.create(DATA_SOURCE_CONFIG)
                // 全局配置
                .globalConfig((scanner, builder) -> builder.author(scanner.apply("xuyanqing"))
                        .fileOverride()
                        .enableSwagger()
                        .dateType(DateType.ONLY_DATE)
                        .outputDir("D:\\testspace\\code_generator")
                )
                // 包配置（父包名）、模块名称
                .packageConfig((scanner, builder) -> builder.parent(scanner.apply("org.topnet.nqi.intranet"))
                        .moduleName("audit")
                        .entity("entity")
                        .mapper("dao")
                        .service("service")
                        .serviceImpl("service.impl")
                )
                // 策略配置
//                .strategyConfig((scanner, builder) -> builder.addInclude(scanner.apply("请输入表名，多个表名用,隔开")).addFieldPrefix("nqi_"))
                .strategyConfig((scanner, builder) -> builder.addInclude(scanner.apply("dm_area")).addFieldPrefix("nqi_"))
                *//*
                    模板引擎配置，默认 Velocity 可选模板引擎 Beetl 或 Freemarker 或 Enjoy
                   .templateEngine(new BeetlTemplateEngine())
                   .templateEngine(new FreemarkerTemplateEngine())
                   .templateEngine(new EnjoyTemplateEngine())
                 *//*
                .templateEngine(new FreemarkerTemplateEngine())
                .execute();*/
}
